import numpy as np
from numpy.linalg import *

group = np.array([
    [-2, 0, 0],
    [1, 1, 0],
    [-3, 0, 1]
])

group1 = np.array([
    [1, 2],
    [3, 4],
    [22, 224],
])

group2 = np.array([
    [11],
    [22],
    [33],
])
a = [
    [1, 2],
    [3, 4]
]
b = [
    [1, 0],
    [0, 1]
]

# print(group1.tolist())


# print(np.hstack((group,group1,group2)))
# print(np.array(a)-(np.array(b)))
# print(np.array(a).dot(np.array(b)))
# for i in range(1,3):
#     print(i)
# c=[1, 22, -3]
# for index,val in enumerate(c):
#     print(index,val)
from numpy import  *
def createDataSet():
    """
    创建数据集，‘numpy.ndarray’类型
    :return: 训练集 训练集对应类别集合
    """
    group = array([
        [1.0, 1.1],
        [1., 1.],
        [0., 0.],
        [0, 0.1]
    ])

    labels = ['A', 'A', 'B', 'B']

    return group, labels

# 返回数据集行数



